/*
 * This file is part of aion-engine <aion-engine.com>
 *
 * aion-engine is private software: you can redistribute it and or modify
 * it under the terms of the GNU Lesser Public License as published by
 * the Private Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * aion-engine is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser Public License for more details.
 *
 * You should have received a copy of the GNU Lesser Public License
 * along with aion-engine.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.aionengine.gameserver.dataholders;

import com.aionengine.gameserver.model.templates.challenge.ChallengeQuestTemplate;
import com.aionengine.gameserver.model.templates.challenge.ChallengeTaskTemplate;

import javax.xml.bind.Unmarshaller;
import javax.xml.bind.annotation.*;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * @author ViAl
 */
@XmlAccessorType(XmlAccessType.FIELD)
@XmlType(name = "", propOrder = {"task"})
@XmlRootElement(name = "challenge_tasks")
public class ChallengeData {

    protected List<ChallengeTaskTemplate> task;

    @XmlTransient
    protected Map<Integer, ChallengeTaskTemplate> tasksById = new HashMap<Integer, ChallengeTaskTemplate>();

    void afterUnmarshal(Unmarshaller u, Object parent) {
        for (ChallengeTaskTemplate t : task) {
            tasksById.put(t.getId(), t);
        }
        task.clear();
        task = null;
    }

    public Map<Integer, ChallengeTaskTemplate> getTasks() {
        return this.tasksById;
    }

    public ChallengeTaskTemplate getTaskByTaskId(int taskId) {
        return tasksById.get(taskId);
    }

    public ChallengeTaskTemplate getTaskByQuestId(int questId) {
        for (ChallengeTaskTemplate ct : tasksById.values()) {
            for (ChallengeQuestTemplate cq : ct.getQuests())
                if (cq.getId() == questId)
                    return ct;
        }
        return null;
    }

    public ChallengeQuestTemplate getQuestByQuestId(int questId) {
        for (ChallengeTaskTemplate ct : tasksById.values()) {
            for (ChallengeQuestTemplate cq : ct.getQuests())
                if (cq.getId() == questId)
                    return cq;
        }
        return null;
    }

    public int size() {
        return this.tasksById.size();
    }
}
